/*
 *------------------------------------------------------------------------------
 * @File      :    ei_mai_nna.h
 * @Date      :    2022-2-9
 * @Author    :    lomboswer <lomboswer@lombotech.com>
 * @Brief     :    Media InterLICPLATE for MADP(Media Development Platform).
 *
 * Copyright (C) 2020-2021, LomboTech Co.Ltd. All rights reserved.
 *------------------------------------------------------------------------------
 */

#ifndef __EI_NNA_PETS_DET_DEFINE_H__
#define __EI_NNA_PETS_DET_DEFINE_H__

#ifdef __cplusplus
extern "C" {
#endif /* __cplusplus */

#include "ei_nna_define.h"

#define DEFAULT_NNA_PETS_DET_CONFIG_MAX_DET_CNT (30)
#define DEFAULT_NNA_PETS_DET_CONFIG_IOU_THRESH (45)
#define DEFAULT_NNA_PETS_DET_CONFIG_CONF_THRESH (40)
#define DEFAULT_NNA_PETS_DET_CONFIG_DET_MODEL_TYPE (0)
// #define DEFAULT_NNA_PETS_DET_CONFIG_DOG_DET_SIZE_RANGE0 (0)


typedef struct eiNNA_PET_S {
	EI_S32 s32Class;
	EI_S32 s32Rect[4];
	EI_FLOAT f32Score;
//	EI_FLOAT f32Kpts[8];
} EI_NNA_PET_S;


typedef struct eiNNA_PETS_S {
	EI_U32 u32Num;
	EI_NNA_PET_S stPetsRect[DEFAULT_NNA_PETS_DET_CONFIG_MAX_DET_CNT];
} EI_NNA_PETS_S;

typedef struct eiNNA_PET_CONFIG_S {
	EI_CHAR *ModelRootPath;
	EI_S32 s32ModelType; //default 0
	EI_S32 s32Width;
	EI_S32 s32Height;
	EI_S32 s32MaxDetCnt;
	EI_S32 s32IouThresh;
	EI_S32 s32ConfThresh;
	// EI_S32 s32Roi[4];

	EI_NNA_AX_FREQ_S stFreq;
} EI_NNA_PET_DET_CONFIG_S;

typedef EI_S32 NNA_PETS_CHN;
typedef EI_NNA_PETS_S EI_NNA_DST_PETS_S;
typedef EI_NNA_PET_DET_CONFIG_S EI_NNA_PET_CONFIG_S;

#ifdef __cplusplus
}

#endif
#endif
